#include <iostream>
#include <fstream>
#include <string>
#include <vector>
#include <sys/stat.h>
#include "../source/server.hpp"

#define DEFALT_TIMEOUT 10

std::unordered_map<int, std::string> _statu_msg = {
    {100,  "Continue"},
    {101,  "Switching Protocol"},
    {102,  "Processing"},
    {103,  "Early Hints"},
    {200,  "OK"},
    {201,  "Created"},
    {202,  "Accepted"},
    {203,  "Non-Authoritative Information"},
    {204,  "No Content"},
    {205,  "Reset Content"},
    {206,  "Partial Content"},
    {207,  "Multi-Status"},
    {208,  "Already Reported"},
    {226,  "IM Used"},
    {300,  "Multiple Choice"},
    {301,  "Moved Permanently"},
    {302,  "Found"},
    {303,  "See Other"},
    {304,  "Not Modified"},
    {305,  "Use Proxy"},
    {306,  "unused"},
    {307,  "Temporary Redirect"},
    {308,  "Permanent Redirect"},
    {400,  "Bad Request"},
    {401,  "Unauthorized"},
    {402,  "Payment Required"},
    {403,  "Forbidden"},
    {404,  "Not Found"},
    {405,  "Method Not Allowed"},
    {406,  "Not Acceptable"},
    {407,  "Proxy Authentication Required"},
    {408,  "Request Timeout"},
    {409,  "Conflict"},
    {410,  "Gone"},
    {411,  "Length Required"},
    {412,  "Precondition Failed"},
    {413,  "Payload Too Large"},
    {414,  "URI Too Long"},
    {415,  "Unsupported Media Type"},
    {416,  "Range Not Satisfiable"},
    {417,  "Expectation Failed"},
    {418,  "I'm a teapot"},
    {421,  "Misdirected Request"},
    {422,  "Unprocessable Entity"},
    {423,  "Locked"},
    {424,  "Failed Dependency"},
    {425,  "Too Early"},
    {426,  "Upgrade Required"},
    {428,  "Precondition Required"},
    {429,  "Too Many Requests"},
    {431,  "Request Header Fields Too Large"},
    {451,  "Unavailable For Legal Reasons"},
    {501,  "Not Implemented"},
    {502,  "Bad Gateway"},
    {503,  "Service Unavailable"},
    {504,  "Gateway Timeout"},
    {505,  "HTTP Version Not Supported"},
    {506,  "Variant Also Negotiates"},
    {507,  "Insufficient Storage"},
    {508,  "Loop Detected"},
    {510,  "Not Extended"},
    {511,  "Network Authentication Required"}
};

std::unordered_map<std::string, std::string> _mime_msg = {
    {".aac",        "audio/aac"},
    {".abw",        "application/x-abiword"},
    {".arc",        "application/x-freearc"},
    {".avi",        "video/x-msvideo"},
    {".azw",        "application/vnd.amazon.ebook"},
    {".bin",        "application/octet-stream"},
    {".bmp",        "image/bmp"},
    {".bz",         "application/x-bzip"},
    {".bz2",        "application/x-bzip2"},
    {".csh",        "application/x-csh"},
    {".css",        "text/css"},
    {".csv",        "text/csv"},
    {".doc",        "application/msword"},
    {".docx",       "application/vnd.openxmlformats-officedocument.wordprocessingml.document"},
    {".eot",        "application/vnd.ms-fontobject"},
    {".epub",       "application/epub+zip"},
    {".gif",        "image/gif"},
    {".htm",        "text/html"},
    {".html",       "text/html"},
    {".ico",        "image/vnd.microsoft.icon"},
    {".ics",        "text/calendar"},
    {".jar",        "application/java-archive"},
    {".jpeg",       "image/jpeg"},
    {".jpg",        "image/jpeg"},
    {".js",         "text/javascript"},
    {".json",       "application/json"},
    {".jsonld",     "application/ld+json"},
    {".mid",        "audio/midi"},
    {".midi",       "audio/x-midi"},
    {".mjs",        "text/javascript"},
    {".mp3",        "audio/mpeg"},
    {".mpeg",       "video/mpeg"},
    {".mpkg",       "application/vnd.apple.installer+xml"},
    {".odp",        "application/vnd.oasis.opendocument.presentation"},
    {".ods",        "application/vnd.oasis.opendocument.spreadsheet"},
    {".odt",        "application/vnd.oasis.opendocument.text"},
    {".oga",        "audio/ogg"},
    {".ogv",        "video/ogg"},
    {".ogx",        "application/ogg"},
    {".otf",        "font/otf"},
    {".png",        "image/png"},
    {".pdf",        "application/pdf"},
    {".ppt",        "application/vnd.ms-powerpoint"},
    {".pptx",       "application/vnd.openxmlformats-officedocument.presentationml.presentation"},
    {".rar",        "application/x-rar-compressed"},
    {".rtf",        "application/rtf"},
    {".sh",         "application/x-sh"},
    {".svg",        "image/svg+xml"},
    {".swf",        "application/x-shockwave-flash"},
    {".tar",        "application/x-tar"},
    {".tif",        "image/tiff"},
    {".tiff",       "image/tiff"},
    {".ttf",        "font/ttf"},
    {".txt",        "text/plain"},
    {".vsd",        "application/vnd.visio"},
    {".wav",        "audio/wav"},
    {".weba",       "audio/webm"},
    {".webm",       "video/webm"},
    {".webp",       "image/webp"},
    {".woff",       "font/woff"},
    {".woff2",      "font/woff2"},
    {".xhtml",      "application/xhtml+xml"},
    {".xls",        "application/vnd.ms-excel"},
    {".xlsx",       "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"},
    {".xml",        "application/xml"},
    {".xul",        "application/vnd.mozilla.xul+xml"},
    {".zip",        "application/zip"},
    {".3gp",        "video/3gpp"},
    {".3g2",        "video/3gpp2"},
    {".7z",         "application/x-7z-compressed"}
};

//字符串分割函数，将src字符串按照sep字符进行分割，得到各个子串放到arry中，最后返回子串的数量
static size_t Split(const std::string &src, const std::string &sep, std::vector<std::string> *arry)
{
    size_t offset=0;
    //offset是查找的其实位置，如果有10个字符，返回就是0~9，offset==10表示已经越界
    while(offset<src.size())
    {
        //在src字符串偏移量offset的地方开始，向后查找sep字符/子串，返回查找到的位置
        size_t pos=src.find(sep,offset);
        
        if(pos==std::string::npos) //没有找到特定的字符
        {
            //将剩余的部分当做一个字符，放入arry中
            if(pos==src.size()) break;
            arry->push_back(src.substr(offset));
            return arry->size();
        }

        if(pos==offset)
        {
            offset=pos+sep.size(); 
            continue; //当前字符串是一个空的，没有内容
        }
        
        arry->push_back(src.substr(offset,pos-offset));
        offset=pos+sep.size();
    }
    return arry->size();
}

//读取文件内容，将读取文件filename的内容放到一个Buffer中
static bool ReadFile(const std::string &filename, std::string *buf)
{
    std::ifstream ifs(filename, std::ios::binary); //打开文件
    if(ifs.is_open()==false) //打开文件失败
    {
        printf("OPEN %s FILE FAILED!",filename.c_str());
        return false;
    }

    size_t fsize=0;
    ifs.seekg(0,ifs.end); //跳转读写位置末尾
    fsize=ifs.tellg(); //获取当前读写位置相对于起始位置的偏移量，从末尾偏移刚好就是文件大小
    ifs.seekg(0,ifs.beg); //跳转到起始位置
    buf->resize(fsize); //开辟文件大小的空间
    ifs.read(&(*buf)[0],fsize);
    
    if(ifs.good()==false)
    {
        printf("READ %s FILE FAILED!",filename.c_str());
        ifs.close();
        return false;
    }

    ifs.close();
    return true;
}   

//向文件写入数据
static bool WriteFile(const std::string &filename, const std::string &buf)
{
    std::ofstream ofs(filename,std::ios::binary|std::ios::trunc); //打开文件，写入数据，覆盖原来的内容
    if(ofs.is_open()==false)
    {
        printf("OPEN %s FILE FAILED!",filename.c_str());
        return false;
    }

    ofs.write(buf.c_str(),buf.size()); //向文件中写入数据
    if(ofs.good()==false)
    {
        ERR_LOG("WRITE %s FILE FAILED!",filename.c_str());
        ofs.close();
        return false;
    }

    ofs.close();
    return true;
}


//URL编码
static std::string UrlEncode(const std::string url, bool convert_space_to_plus)
{
    std::string res;
    for(auto &c:url)
    {
        if(c=='.'||c=='-'||c=='_'||c=='~'||isalnum(c))
        {
            res+=c;
            continue;
        }
        if(c==' '&&convert_space_to_plus==true)
        {
            res+='+';
            continue;
        }
        //剩下的字符都是需要编码为%HH格式
        char tmp[4]={0};
        //snprintf 与 printf比较类似，都是格式化字符串，只不过一个是打印，一个是放到一块空间中
        snprintf(tmp,4,"%%%02x",c);
        res+=tmp;
    }
    return res;
}

//将字符转换为数字
static char HEXTOI(char c)
{
    if(c>='0'&&c<='9')
    {
        return c-'0';
    }
    else if(c>='a'&&c<='z')
    {
        return c-'a'+10;
    }
    else if(c>='A'&&c<='Z')
    {
        return c-'A'+10;
    }
    return -1;
}

//URL解码
static std::string UrlDecode(const std::string url, bool convert_plus_to_space)
{
    //遇到了%，就将紧随其后的2个字符，转换为数字，第一个数字左移4位，然后加上第二个数字，+ -> 2b  %2b -> 2<<4+11
    std::string res;
    for(int i=0;i<url.size();i++)
    {
        if (url[i] == '+' && convert_plus_to_space == true) 
        {
            res += ' ';
            continue;
        }
        if(url[i]=='%')
        {
            char v1=HEXTOI(url[i+1]);
            char v2=HEXTOI(url[i+2]);
            char v=(v1<<4)+v2;
            res+=v;
            i+=2;
            continue;
        }
        res+=url[i];
    }
    return res;
}

//响应状态码的描述符信息获取
static std::string StatuDesc(int statu)
{
    auto it=_statu_msg.find(statu);
    if(it!=_statu_msg.end())
    {
        return it->second;
    }
    return "Unknow";
}

//根据文件后缀名获取文件mime
static std::string ExtMime(const std::string &filename)
{
    // a.b.txt  先获取文件扩展名
    size_t pos = filename.find_last_of('.');
    if (pos == std::string::npos) 
    {
        return "application/octet-stream";
    }
    
    //根据扩展名，获取mime
    std::string ext = filename.substr(pos);
    auto it = _mime_msg.find(ext);
    if (it == _mime_msg.end()) 
    {
        return "application/octet-stream";
    }
    return it->second;
}

//判断一个文件是否是一个目录
static bool IsDirectory(const std::string &filename)
{
    struct stat st;
    int ret=stat(filename.c_str(),&st);
    if(ret<0)
    {
        return false;
    }
    return S_ISDIR(st.st_mode);
}

//判断一个文件是否是普通文件
static bool IsRegular(const std::string &filename)
{
    struct stat st;
    int ret=stat(filename.c_str(),&st);
    if(ret<0)
    {
        return false;
    }
    return S_ISREG(st.st_mode);
}

//http请求的资源路径的有效性判断
static bool ValidPath()
{

}

int main()
{
    std::cout<<StatuDesc(404)<<std::endl;
    std::cout<<ExtMime("a.png")<<std::endl;
    std::cout<<IsRegular("regex.cpp")<<std::endl;
    std::cout<<IsRegular("XXXXX")<<std::endl;
    std::cout<<IsDirectory("regex.cpp")<<std::endl;
    std::cout<<IsDirectory("testdir")<<std::endl;

    /*
    std::string str="C++";
    //std::string str="/login?user=bite&passwd=C++";
    //std::string res;
    std::string res=UrlEncode(str,false);
    std::string tmp=UrlDecode(res,false);
    std::cout<<"["<<res<<"]"<<std::endl;
    std::cout<<"["<<tmp<<"]"<<std::endl;
    */

    /*
    std::string str="/index.html?word=c++";
    std::string res;
    for(auto &c:str)
    {
        char tmp[4]={0};
        snprintf(tmp,4,"%%%x",c);
        std::cout<<tmp<<std::endl;
    }*/

    /*
    std::string str="abc,,,,,bcd,,,def,,";
    std::vector<std::string> arry;
    Split(str,",",&arry);
    for(auto &s:arry)
    {
        std::cout<<"["<<s<<"]\n";
    }

    std::string buf;
    bool ret=ReadFile("./bind.cpp",&buf);
    if(ret==false)
    {
        std::cout<<"ReadFile false"<<std::endl;
        return false;
    }
    std::cout<<buf<<std::endl;

    ret=WriteFile("./test.txt",buf);
    if(ret==false)
    {
        return false;
    }
    //std::cout<<buf.GetReadPosition()<<std::endl;
    */

    return 0;
}